# -*- coding: utf-8 -*-
# @Time    : 2020/6/5
# @Author  : nanjiang.xie
# @File    : mysqlLib.py

"""
封装mysql操作关键字
"""

import pymysql


def execute_mysql(host, user, passwd, cmd, db, port="3306", charset="utf8"):
    """执行MySQL语句
    - :param host: 【必填】数据库服务器名称或IP，类型：string
    - :param user: 【必填】mysql用户名，类型：string
    - :param passwd: 【必填】mysql密码，类型：string
    - :param cmd: 【必填】mysql命令，类型: string
    - :param db: 【必填】mysql的database，类型：string
    - :param port: 【选填】mysql的port，类型：string，默认是3306
    - :param charset: 【选填】mysql的编码方式，类型：string，默认是utf8
    - :return: 返回命令执行结果

    - 举例：
    | Exuecute Mysql | host=47.103.204.228 | user=root | passwd=root | cmd=select * from xxx | db, database=xxx |
    """

    # 创建一个连接对象，再使用创建游标
    con = pymysql.connect(host='47.103.204.228',  user='root', passwd='nanjiang.xie', db='test',  port=3306, charset='utf8')
    cursor = con.cursor()

    # 执行一个SQL语句
    cmd = ('select * from test')
    cursor.execute(cmd)

    # 从游标中取出所有记录放到一个序列中并关闭游标
    result = cursor.fetchone()
    # print(result)
    cursor.close()
    con.close()
    return result
